

package com.hazelcast.client.impl.protocol.codec;

import com.hazelcast.client.impl.protocol.ClientMessage;
import com.hazelcast.client.impl.protocol.Generated;
import com.hazelcast.client.impl.protocol.codec.builtin.ListMultiFrameCodec;
import com.hazelcast.client.impl.protocol.codec.builtin.StringCodec;
import com.hazelcast.client.impl.protocol.codec.custom.VectorPairCodec;
import com.hazelcast.client.impl.protocol.codec.custom.VectorSearchOptionsCodec;
import com.hazelcast.client.impl.protocol.codec.custom.VectorSearchResultCodec;

import static com.hazelcast.client.impl.protocol.ClientMessage.*;
import static com.hazelcast.client.impl.protocol.codec.builtin.FixedSizeTypesCodec.*;

/*
 * This file is auto-generated by the Hazelcast Client Protocol Code Generator.
 * To change this file, edit the templates or the protocol
 * definitions on the https://github.com/hazelcast/hazelcast-client-protocol
 * and regenerate it.
 */

/**
 * Returns the VectorDocuments closest to the given vector.
 */
@SuppressWarnings("unused")
@Generated("7567d98a9c5b86950c1847370c2b0ee4")
public final class VectorCollectionSearchNearVectorCodec {
    //hex: 0x240800
    public static final int REQUEST_MESSAGE_TYPE = 2361344;
    //hex: 0x240801
    public static final int RESPONSE_MESSAGE_TYPE = 2361345;
    private static final int REQUEST_INITIAL_FRAME_SIZE = PARTITION_ID_FIELD_OFFSET + INT_SIZE_IN_BYTES;
    private static final int RESPONSE_INITIAL_FRAME_SIZE = RESPONSE_BACKUP_ACKS_FIELD_OFFSET + BYTE_SIZE_IN_BYTES;

    private VectorCollectionSearchNearVectorCodec() {
    }

    public static ClientMessage encodeRequest(java.lang.String name, com.hazelcast.vector.VectorValues vectors, com.hazelcast.vector.SearchOptions options) {
        ClientMessage clientMessage = ClientMessage.createForEncode();
        clientMessage.setRetryable(true);
        clientMessage.setOperationName("VectorCollection.SearchNearVector");
        ClientMessage.Frame initialFrame = new ClientMessage.Frame(new byte[REQUEST_INITIAL_FRAME_SIZE], UNFRAGMENTED_MESSAGE);
        encodeInt(initialFrame.content, TYPE_FIELD_OFFSET, REQUEST_MESSAGE_TYPE);
        encodeInt(initialFrame.content, PARTITION_ID_FIELD_OFFSET, -1);
        clientMessage.add(initialFrame);
        StringCodec.encode(clientMessage, name);
        ListMultiFrameCodec.encode(clientMessage, vectors, VectorPairCodec::encode);
        VectorSearchOptionsCodec.encode(clientMessage, options);
        return clientMessage;
    }

    public static VectorCollectionSearchNearVectorCodec.RequestParameters decodeRequest(ClientMessage clientMessage) {
        ClientMessage.ForwardFrameIterator iterator = clientMessage.frameIterator();
        RequestParameters request = new RequestParameters();
        //empty initial frame
        iterator.next();
        request.name = StringCodec.decode(iterator);
        request.vectors = ListMultiFrameCodec.decode(iterator, VectorPairCodec::decode);
        request.options = VectorSearchOptionsCodec.decode(iterator);
        return request;
    }

    public static ClientMessage encodeResponse(java.lang.Iterable<com.hazelcast.vector.SearchResult<com.hazelcast.internal.serialization.Data, com.hazelcast.internal.serialization.Data>> result) {
        ClientMessage clientMessage = ClientMessage.createForEncode();
        ClientMessage.Frame initialFrame = new ClientMessage.Frame(new byte[RESPONSE_INITIAL_FRAME_SIZE], UNFRAGMENTED_MESSAGE);
        encodeInt(initialFrame.content, TYPE_FIELD_OFFSET, RESPONSE_MESSAGE_TYPE);
        clientMessage.add(initialFrame);

        ListMultiFrameCodec.encode(clientMessage, result, VectorSearchResultCodec::encode);
        return clientMessage;
    }

    /**
     * Zero or more VectorSearchResult values.
     */
    public static java.util.List<com.hazelcast.vector.impl.DataSearchResult> decodeResponse(ClientMessage clientMessage) {
        ClientMessage.ForwardFrameIterator iterator = clientMessage.frameIterator();
        //empty initial frame
        iterator.next();
        return ListMultiFrameCodec.decode(iterator, VectorSearchResultCodec::decode);
    }

    @edu.umd.cs.findbugs.annotations.SuppressFBWarnings({"URF_UNREAD_PUBLIC_OR_PROTECTED_FIELD"})
    public static class RequestParameters {

        /**
         * Name of the Vector Collection.
         */
        public java.lang.String name;

        /**
         * Vector for which closest neighbours should be returned.
         */
        public java.util.List<com.hazelcast.client.impl.protocol.codec.holder.VectorPairHolder> vectors;

        /**
         * Search options.
         */
        public com.hazelcast.vector.SearchOptions options;
    }
}
